#include <iostream>
#include <vector>
using namespace std;


namespace hash_bucket
{
	template<class T>
	struct HashNode
	{
		T _data;
		HashNode<T>* _next;
		HashNode(const T& data)
			:_data(data)
			,_next(nullptr)
		{}
	};

   
    // K 为 T 中key的类型
    // T 可能是键值对，也可能是K
    // KeyOfT: 从T中提取key
    // Hash将key转化为整形，因为哈市函数使用除留余数法
	template<class K, class T, class KeyOfT, class Hash>
	class HashTable
	{

		typedef HashNode<T> Node;
	public:
		HashTable()
		{
			_tables.resize(10, nullptr);
		}

        // 哈希桶的销毁
		~HashTable() {
            for(int i = 0; i < _tables.size(); i++) {
                Node* cur = _tables[i];
                while (cur) {
                    Node* next = cur->_next;
                    delete cur;
                    cur = next;
                }
                _tables[i] = nullptr;
            }
        }

        // 插入值为data的元素，如果data存在则不插入
		bool Insert(const T& data) {
            // 显示判断是不是存在
            KeyOfT kot;
            Hash hash;
            K key = kot(data);
            int pos = hash(key) % _tables.size();
            Node* cur = _tables[pos];
            while (cur) {
                Node* next = cur->_next;
                if(kot(cur->_data) == key) {
                    return false;
                }
                cur = next;
            }
            Node* newnode = new Node(data);
            newnode->_next = _tables[pos];
            _tables[pos] = newnode;
            ++_n;
            if((double)_n / _tables.size() > 1.0) {
                _Resize();
            }
            return true;
        }

        // 在哈希桶中查找值为key的元素，存在返回true否则返回false﻿
		bool Find(const K& key) {
            KeyOfT kot;
            Hash hash;
            int pos = hash(key) % _tables.size();
            Node* cur = _tables[pos];
            while(cur) {
                if(kot(cur->_data) == key) {
                    return true;
                }
                cur = cur->_next;
            }
            return false;
        }

        // 哈希桶中删除key的元素，删除成功返回true，否则返回false
		bool Erase(const K& key) {
            KeyOfT kot;
            Hash hash;
            int pos = hash()(key) % _tables.size();
            Node* cur = _tables[pos];
            Node* pre = nullptr;
            while (cur) {
                if(kot(cur->_data) == key) {
                    if(pre) {
                        pre->_next = cur->_next;
                    }else {
                        _tables[pos] = cur->_next;
                    }
                    delete cur;
                    --_n;
                    return true;
                }
                pre = cur;
                cur = cur->_next;
            }
            return false;
        }
	private:
        void _Resize() {
            int newsize = _tables.size() * 2;
            vector<Node*> _newtables(newsize, nullptr);
            KeyOfT kot;
            Hash hash;
            for(auto& head : _tables) {
                Node* cur = head;
                while (cur) {
                    Node* next = cur->_next;
                    int pos = hash(kot(cur->_data)) % newsize;
                    cur->_next = _newtables[pos];
                    _newtables[pos] = cur;
                    cur = next;
                }
            }
            _tables.swap(_newtables);
        }
		vector<Node*> _tables;  // 指针数组
		size_t _n = 0;			// 表中存储数据个数
	};
}